import numpy as np
import cv2






class DataSet:
    def __init__(self,batch_size:int,IsRandomChoose:bool) -> None:
        self.batch_size = batch_size
        self.IsRandomChoose = IsRandomChoose
        pass
    
    def LoadData(self,shape:tuple = None)->(np.matrix,np.matrix):
        """
        返回训练集和测试集
        """
        pass
    
       


def ImageDataSet(DataSet):
    def __init__(self,batch_size:int,IsRandomChoose:bool) -> None:
        self.batch_size = batch_size
        self.IsRandomChoose = IsRandomChoose
    pass